import matplotlib.pyplot as plt
from plot.random_walk import RandomWalk

rw = RandomWalk(5000)
rw.fill_walk()

point_numbers = list(range(rw.num_points))
# 点图
plt.scatter(rw.x_values, rw.y_values, s=10, edgecolors="none", c=point_numbers, cmap=plt.cm.Blues)
# 折线图
# plt.plot(rw.x_values,rw.y_values,linewidth=5)

# 突出起点和终点
plt.scatter(0, 0, s=40, c="green", edgecolors="none")
plt.scatter(rw.x_values[-1], rw.y_values[-1], c="red", edgecolors="none", s=40)

# 隐藏坐标轴
# plt.axes().get_xaxis().set_visible(False)
# plt.axes().get_yaxis().set_visible(False)

# 设置绘图窗口的尺寸
plt.figure(figsize=[10, 6], dpi=720)

plt.show()
